#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <stdio.h>
#include <pthread.h>
#include <time.h>
#include <string.h>

#include "functions.c"

pthread_mutex_t mux_cola1 = PTHREAD_MUTEX_INITIALIZER;
pthread_mutex_t mux_velocidad = PTHREAD_MUTEX_INITIALIZER;
Cola cola;
maquina maquina1;
maquina maquina2;
int velocidad;
int aux=0;
void initParameters(char *argv[]) {
	cola.cantidad = 0;
	cola.longitud = atoi(argv[1]);
	velocidad = atoi(argv[2]);
	maquina1.estado = -1;
	maquina2.estado = -1;
	maquina1.start = time(NULL);
	maquina2.start = time(NULL);
	printf("\nVelocidad Llenar Botellas: %d", velocidad);
	printf("\nLongitud cola 1: %d\n", cola.longitud);
}

void *confVelocidad(void * arg) {
	while (maquina2.estado) {
		int cambioVelocidad = 0;
		scanf("%d", &cambioVelocidad);

		if (pthread_mutex_lock(&mux_velocidad) == 0) {
			velocidad = cambioVelocidad;
		}
		if (pthread_mutex_unlock(&mux_velocidad) != 0) {
			error("Error Desbloquer Velocidad");
		}
	}
	pthread_exit(NULL);
}

int anyadirBotella() {
	if (cola.longitud != cola.cantidad) {
		if (pthread_mutex_lock(&mux_cola1) == 0) {
			cola.cantidad = cola.cantidad + 1;
		}
		if (pthread_mutex_unlock(&mux_cola1) != 0) {
			error("Error Desbloquer Cola1");
		}
		return 1;
	} else {
		return 0;
	}
}

void *maquina12(void * arg) {
	int fd, fd2, n, sin_size;
	struct sockaddr_in server;
	struct sockaddr_in client;
	if ((fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
		error("Error en Socket Server M2");
	}
	server.sin_family = AF_INET;
	server.sin_port = htons(PORTM1M2);
	server.sin_addr.s_addr = INADDR_ANY;
	bzero(&(server.sin_zero), 8);

	int reuse = 1;
	setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char*) &reuse, sizeof(reuse));

	if (bind(fd, (struct sockaddr*) &server, sizeof(struct sockaddr)) == -1) {
		error("Error en Socket Server M2 (BIND)");
	}

	if (listen(fd, BACKLOG) == -1) {
		error("Error en Socket Server M2 (LISTEN)");
	}

	while (maquina1.estado) {
		sin_size = sizeof(struct sockaddr_in);
		if ((fd2 = accept(fd, (struct sockaddr *) &client, &sin_size)) == -1) {
			error("Error en Socket Server M2 (ACCEPT)");
		}

		n = recv(fd2, &maquina1, sizeof(maquina1), 0);
		if (n < 0)
			error("Error Recibiendo Mensaje Server M2 (RECV)");
		int resposta = 1;
		if (maquina1.estado == 0) {
			n = send(fd2, &resposta, sizeof(resposta), 0);
			if (n < 0)
				error("Error Enviando Mensaje Server M2 (SEND)");
			traceM2("Stop Maquina 1", maquina1.start, velocidad, cola);
			pthread_cond_signal(&cola.vacia);
			pthread_exit(NULL);
		} else {
			int add = anyadirBotella(maquina1.start);
			n = send(fd2, &add, sizeof(add), 0);
			if (n < 0)
				error("Error Enviando Mensaje Server M2 (SEND)");

			if (!add) {
				//Cola Llena, esperar a que este disponible
				pthread_mutex_lock(&mux_cola1);
				traceM2("Esperar Cola1", maquina1.start, velocidad, cola);
				pthread_cond_wait(&cola.llena, &mux_cola1);
				pthread_mutex_unlock(&mux_cola1);
			} else if (cola.cantidad > 0) {
				pthread_cond_signal(&cola.vacia);
			}

		}
		close(fd2);
	}
	pthread_cond_signal(&cola.vacia);
	pthread_exit(NULL);
}

void *maquina23(void * arg) {
	while (maquina2.estado || cola.cantidad > 0) {
		if (cola.cantidad > 0) {
			if (aux==0){
				headerM2M3();
				aux=1;
			}

			traceM2("LLenando Botella", maquina2.start, velocidad, cola);
			if (pthread_mutex_lock(&mux_velocidad) == 0) {
				usleep(1000000 / velocidad);
			}
			if (pthread_mutex_unlock(&mux_velocidad) != 0) {
				error("Error Desbloquer Velocidad");
			}
			if (enviarMaquina3(0)) {
				if (pthread_mutex_lock(&mux_cola1) == 0) {
					cola.cantidad = cola.cantidad - 1;
					traceM2("Poniendo Botella", maquina2.start, velocidad, cola);
				}
				if (pthread_mutex_unlock(&mux_cola1) != 0) {
					error("Error Desbloquer Cola1");
				}

				//Cola 1 Disponible
				if (cola.longitud != cola.cantidad) {
					pthread_cond_signal(&cola.llena);
				}

				//Cola 1 Vacia
				if (maquina1.estado && cola.cantidad == 0) {
					pthread_mutex_lock(&mux_cola1);
					traceM2("Cola1 Vacia   ", maquina1.start, velocidad, cola);
					pthread_cond_wait(&cola.vacia, &mux_cola1);
					pthread_mutex_unlock(&mux_cola1);
				}

				traceM2("Botella Llena", maquina2.start, velocidad, cola);
			} else {
				traceM2("Cola2 Llena  ", maquina2.start, velocidad, cola);
			}

		} else if (maquina1.estado == 0) {
			if (enviarMaquina3(1)) {
				traceM2("Apagando Maquina", maquina2.start, velocidad, cola);
				pthread_exit(NULL);
			}
		}
	}

}

int enviarMaquina3(int stop) {
	int fd, n;
	struct sockaddr_in server;
	server.sin_family = AF_INET;
	server.sin_port = htons(PORTM2M3);
	server.sin_addr.s_addr = inet_addr(LOCALHOST);
	bzero(&(server.sin_zero), 8);

	if ((fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
		error("Error en Socket Cliente M2");
	}

	if (connect(fd, (struct sockaddr *) &server, sizeof(struct sockaddr)) == -1) {
		error("Error en Socket Cliente M2 (CONNECT)");
	}

	maquina2.start = maquina1.start;

	//Preguntar a M2 si puede añadir botella o avisar si para M1
	if (stop) {
		maquina2.estado = 0;
	} else {
		maquina2.estado = 1;
	}
	n = send(fd, &maquina2, sizeof(maquina2), 0);
	if (n < 0)
		error("Error Enviando Mensaje Cliente M2 (SEND)");

	//Respuesta M2
	int resposta;
	n = recv(fd, &resposta, sizeof(resposta), 0);
	if (n < 0)
		error("Error Recibiendo Mensaje Cliente M2 (RECV)");
	close(fd);

	return resposta;
}

int main(int argc, char **argv) {

	if (argc != 3) {
		error("Argumentos no validos: <longitug cola1> <velocidad de llenar botellas>");
	}

	initParameters(argv);

	pthread_t threadM12;
	pthread_t threadM23;
	pthread_t threadConfVelocidad;

	pthread_create(&threadM12, NULL, maquina12, NULL);
	pthread_create(&threadM23, NULL, maquina23, NULL);
	pthread_create(&threadConfVelocidad, NULL, confVelocidad, NULL);

	pthread_cond_init(&cola.llena, NULL);
	pthread_cond_init(&cola.vacia, NULL);

	(void) pthread_join(threadM12, NULL);
	(void) pthread_join(threadM23, NULL);

	pthread_cancel(threadConfVelocidad);
	pthread_mutex_destroy(&mux_velocidad);
	pthread_mutex_destroy(&mux_cola1);
}

